Create a mask according to the value of a numpy array